import fitz  # PyMuPDF
import os
def extract_first_pages(pdf_path, num_pages=10):
    # 打开PDF文件
    doc = fitz.open(pdf_path)
    
    # 创建一个新的PDF文档来存储前10页
    new_doc = fitz.open()
    
    # 遍历前10页并添加到新文档中
    for i in range(min(num_pages, len(doc))):
        page = doc.load_page(i)  # 加载页面
        new_doc.insert_pdf(doc, from_page=i, to_page=i)  # 将当前页插入新文档
    
    # 保存新文档
    new_pdf_path = '1-127.pdf'
    new_doc.save(new_pdf_path)
    new_doc.close()
    doc.close()
    print(f"前{num_pages}页已保存到 {new_pdf_path}")

# 使用函数
pdf_path = './MyPy/工具/1.pdf'  # 替换为你的PDF文件路径
extract_first_pages(pdf_path,127)
